package com.neusoft.hifly.netty.websocket.handler;

import org.springframework.kafka.core.KafkaTemplate;

import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.stream.ChunkedWriteHandler;

/**
 * 子处理器
 */
public class WebSocketServerInitialzer extends ChannelInitializer<SocketChannel> {

	// 路由
	private String websocketPath;

	// kafka
	private KafkaTemplate<String, Object> kafkaTemplate;

	public WebSocketServerInitialzer(String websocketPath, KafkaTemplate<String, Object> kafkaTemplate) {
		super();
		this.websocketPath = websocketPath;
		this.kafkaTemplate = kafkaTemplate;
	}

	@Override
	protected void initChannel(SocketChannel ch) throws Exception {
		ChannelPipeline pipeline = ch.pipeline();

		// websocket 基于http协议，所以要有http编解码器 服务端用HttpServerCodec
		pipeline.addLast(new HttpServerCodec());
		// 对写大数据流的支持
		pipeline.addLast(new ChunkedWriteHandler());

		/**
		 * 我们通常接收到的是一个http片段，如果要想完整接受一次请求的所有数据，我们需要绑定HttpObjectAggregator，然后我们 就可以收到一个FullHttpRequest-是一个完整的请求信息。
		 * 对httpMessage进行聚合，聚合成FullHttpRequest或FullHttpResponse 几乎在netty中的编程，都会使用到此hanler
		 */
		pipeline.addLast(new HttpObjectAggregator(1024 * 64));

		// ====================== 以上是用于支持http协议 , 以下是支持httpWebsocket   ======================

		/**
		 * websocket 服务器处理的协议，用于指定给客户端连接访问的路由 : /ws
		 * <p>
		 * 本handler会帮你处理一些繁重的复杂的事 会帮你处理握手动作： handshaking（close, ping, pong） ping + pong = 心跳
		 * 对于websocket来讲，都是以frames进行传输的，不同的数据类型对应的frames也不同
		 * <p>
		 * 连接方式：ws://127.0.0.1:8000/webSocket
		 */
		pipeline.addLast(new WebSocketServerProtocolHandler(websocketPath));

		// 自定义的handler
		pipeline.addLast(new WebSocketChatHandler(kafkaTemplate));
	}

}